package org.example.protocol;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.ByteToMessageCodec;
import lombok.extern.slf4j.Slf4j;
import org.example.domain.po.MessagePO;

import java.util.List;

@Slf4j
public class MessageCodec extends ByteToMessageCodec<MessagePO> {

    private static final byte SERIALIZATION = (byte) Algorithm.JAVA.getValue();

    @Override
    protected void encode(ChannelHandlerContext ctx, MessagePO msg, ByteBuf out) throws Exception {
        log.error("开始编码:{}", msg);
        // 设置魔数 4个字节
        out.writeBytes(new byte[]{'N', 'Y', 'I', 'M'});
        // 设置版本号 1个字节
        out.writeByte(1);
        // 设置序列化方式 1个字节 1表示JSON
        out.writeByte(SERIALIZATION);
        // 设置指令类型 1个字节
        out.writeByte(msg.getMessageType());
        // 设置请求序号 4个字节
        out.writeInt(msg.getSequenceId());
        // 为了补齐为16个字节，填充1个字节的数据
        out.writeByte(0xff);

        // 获得序列化后的msg
        byte[] bytes = Algorithm.getByValue(SERIALIZATION).serialize(msg);

        // 获得并设置正文长度 长度用4个字节标识
        out.writeInt(bytes.length);
        // 设置消息正文
        out.writeBytes(bytes);
    }

    @Override
    protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {
        log.info("开始解码");
        // 获取魔数
        int magic = in.readInt();
        // 获取版本号
        byte version = in.readByte();
        // 获得序列化方式
        byte seqType = in.readByte();
        // 获得指令类型
        byte messageType = in.readByte();
        // 获得请求序号
        int sequenceId = in.readInt();
        // 移除补齐字节
        in.readByte();
        // 获得正文长度
        int length = in.readInt();
        // 获得正文
        byte[] bytes = new byte[length];
        RpcSerialize byValue = Algorithm.getByValue(seqType);
        in.readBytes(bytes, 0, length);
        // 根据序列化方式-进行反序列化
        MessagePO message = byValue.deserialize(bytes, MessagePO.getMessageClass(messageType));
        // 将信息放入List中，传递给下一个handler
        out.add(message);

        // 打印获得的信息正文
        log.info("===========魔数===========");
        log.info("{}", magic);
        log.info("===========版本号===========");
        log.info("{}", version);
        log.info("===========序列化方法===========");
        log.info("{}", seqType);
        log.info("===========指令类型===========");
        log.info("{}", messageType);
        log.info("===========请求序号===========");
        log.info("{}", sequenceId);
        log.info("===========正文长度===========");
        log.info("{}", length);
        log.info("===========正文===========");
        log.info("{}", message);
    }
}